package app.modules.menu.persistence.dao;

import app.modules.menu.dto.dal.MenuDalDTO;
import app.modules.menu.dto.dal.MenuDetailDalDTO;
import app.modules.menu.persistence.mapper.MenuMapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import jasmine.framework.common.util.CollectionUtil;
import jasmine.framework.database.mybatisplus.dao.BaseI18nEntityDAO;
import jasmine.security.rbac.model.SecMenu;
import org.springframework.stereotype.Repository;

import javax.annotation.Nullable;
import java.util.Collection;
import java.util.List;

/**
 * @author mh.z
 */
@Repository
public class MenuDAO extends BaseI18nEntityDAO<MenuMapper, SecMenu> {

    /**
     * 查找指定模板下的所有菜单
     *
     * @param templateCode
     * @return
     */
    public List<MenuDalDTO> listMenuDalDTOsByTemplateCode(String templateCode) {
        return baseMapper.listMenuDalDTOsByTemplateCode(templateCode);
    }

    /**
     * 查询指定模板的菜单
     *
     * @param templateId
     * @return
     */
    public List<MenuDetailDalDTO> listMenuDetailDalDTOsByTemplateId(Long templateId) {
        return baseMapper.listMenuDetailDalDTOsByTemplateId(templateId);
    }

    /**
     * 查询指定的菜单
     *
     * @param id
     * @return
     */
    public MenuDetailDalDTO getMenuDetailDalDtoById(Long id) {
        return baseMapper.getMenuDetailDalDtoById(id);
    }

    /**
     * 统计指定菜单的子菜单数
     *
     * @param menuId
     * @return
     */
    public long countChildMenus(Long menuId) {
        LambdaQueryWrapper<SecMenu> wrapper = Wrappers.lambdaQuery();
        wrapper.eq(SecMenu::getParentId, menuId);

        return baseMapper.selectCount(wrapper);
    }

    /**
     * 根据菜单代码统计
     *
     * @param menuCode
     * @param excludeIds
     * @return
     */
    public long countByMenuCode(String menuCode, @Nullable Collection<Long> excludeIds) {
        LambdaQueryWrapper<SecMenu> wrapper = Wrappers.lambdaQuery();

        wrapper.eq(SecMenu::getMenuCode, menuCode);
        wrapper.notIn(CollectionUtil.isNotEmpty(excludeIds), SecMenu::getId, excludeIds);

        return baseMapper.selectCount(wrapper);
    }


}
